def input_username():
    username = input("请输入用户名")
    return username


def input_password():
    password = input("请输入密码")
    return password


def login():
    a = input_username()
    b = input_password()
    if a == "admin" and b == "123456":
        print("登录成功!")
    else:
        print("用户名或密码错误")


login()
